package cn.xaut.数组;

import java.util.Arrays;
import java.util.PriorityQueue;

public class demo88 {

    public void merge(int[] nums1, int m, int[] nums2, int n) {

        int i = 0;
        int p2 = 0;
        while (i < m) {
            if (nums1[i] <= nums2[p2]) {
                i ++;
            } else {
                int t = nums1[i];
                nums1[i] = nums2[p2];
                nums2[p2] = t;
                i ++;
            } 
        }
    }

    // 优先队列
    public void merge2(int[] nums1, int m, int[] nums2, int n) {

        PriorityQueue<Integer> pq = new PriorityQueue<>();
        for (int i = 0; i < m; i++) 
            pq.offer(nums1[i]);

        for (int i = 0; i < n; i++) 
            pq.offer(nums2[i]);

        for (int i = 0; i < m + n; i++) 
            nums1[i] = pq.poll();
    }
    
    // 合并后排序
    public void merge3(int[] nums1, int m, int[] nums2, int n) {

        // todo 将一个数组复制到另一个数组中
        System.arraycopy(nums2, 0, nums1, m, n);
        Arrays.sort(nums1);
    }

    // 双指针/从后往前
    public void merge5(int[] nums1, int m, int[] nums2, int n) {
        
        int p = m + n - 1;
        int p1 = m - 1, p2 = n - 1;
        while (p1 >= 0 && p2 >= 0) 
            nums1[p--] = nums1[p1] >= nums2[p2] ? nums1[p1--] : nums2[p2--];
        
        while (p >= 0)
            nums1[p--] = p2 < 0 ? nums1[p1--] : nums2[p2--];
    }
    
    // 双指针/从前往后（有点小问题）
    public void merge4(int[] nums1, int m, int[] nums2, int n) {

        // 复制数组
        System.arraycopy(nums1, 0, nums1, m, n);
        int p1 = m, p2 = 0;
        int i = 0;
        while (p1 < m + n && p2 < n) {
            if (nums1[p1] <= nums2[p2]) {
                nums1[i] = nums1[p1];
                p1 ++;
            } else {
                nums1[i] = nums2[p2];
                p2 ++;
            } 
            i ++;
        }

        if (p1 == m + n) 
            System.arraycopy(nums2, p2, nums1, i, n - p2);
        else 
            System.arraycopy(nums1, p1, nums1, i, m + n - p1);
    }

    public static void main(String[] args) {

        int[] a1 = new int[]{1, 2, 3, 0, 0, 0};
        int[] a2 = new int[]{2, 5, 6};
        new demo88().merge5(a1, 3, a2, 3);
        System.out.println(Arrays.toString(a1));
    }
}
